package com.lxy.aimin.gateway.config;

import cn.dev33.satoken.context.SaHolder;
import cn.dev33.satoken.reactor.filter.SaReactorFilter;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.util.SaResult;
import com.lxy.aimin.satoken.admin.StpAdminUtil;
import com.lxy.aimin.satoken.user.StpUserUtil;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.lang.reflect.Constructor;

/**
 * [Sa-Token 权限认证] 配置类
 * @author click33
 */
@Configuration
public class SaTokenConfigure {
    // 注册 Sa-Token全局过滤器
    @Bean
    public SaReactorFilter getSaReactorFilter() {
        return new SaReactorFilter()
                // 开放地址
                .addExclude("/test/userLogin")
                .addExclude("/test/adminLogin")
                // 拦截地址
                .addInclude("/**")    /* 拦截全部path */
                // 鉴权方法：每次访问进入
                .setAuth(obj -> {
                    // 登录校验 -- 拦截所有路由，并排除/aimin-auth/test/login 用于开放登录
                    SaRouter.match("/**", "/test/userCheckLogin", r -> StpUserUtil.stpLogic.checkLogin());
                    SaRouter.match("/**", "/test/adminCheckLogin", r -> StpAdminUtil.stpLogic.checkLogin());

                })
                // 异常处理方法：每次setAuth函数出现异常时进入
                .setError(e -> {
                    return SaResult.error(e.getMessage());
                });
    }
}
